gie_inference.h File Reference

Detailed Description

NVIDIA Multimedia API: TensorRT Inference API

Description: This file provides interfaces to load the Caffe model and perform inference with the TensorRT model.

Definition in file gie_inference.h.

Go to the source code of this file.

Data Structures
class	GIE_Context

Macros
#define	GOOGLENET_SINGLE_CLASS   0
#define	GOOGLENET_THREE_CLASS   1
#define	HELNET_THREE_CLASS   2

Variables
struct {
const int   classCnt
float   THRESHOLD [3]
const char *   INPUT_BLOB_NAME
const char *   OUTPUT_BLOB_NAME
const char *   OUTPUT_BBOX_NAME
const int   STRIDE
const int   WORKSPACE_SIZE
float   bbox_output_scales [4]
}	g_pModelNetAttr
struct {
const int   classCnt
float   THRESHOLD [3]
const char *   INPUT_BLOB_NAME
const char *   OUTPUT_BLOB_NAME
const char *   OUTPUT_BBOX_NAME
const int   STRIDE
const int   WORKSPACE_SIZE
float   bbox_output_scales [4]
}	gModelNetAttr [3]

Variable Documentation

float bbox_output_scales[4]

Definition at line 76 of file gie_inference.h.

const int classCnt

Definition at line 69 of file gie_inference.h.

const char* INPUT_BLOB_NAME

Definition at line 71 of file gie_inference.h.

const char* OUTPUT_BBOX_NAME

Definition at line 73 of file gie_inference.h.

const char* OUTPUT_BLOB_NAME

Definition at line 72 of file gie_inference.h.

const int STRIDE

Definition at line 74 of file gie_inference.h.

float THRESHOLD[3]

Definition at line 70 of file gie_inference.h.

const int WORKSPACE_SIZE

Definition at line 75 of file gie_inference.h.